package com.example.service;

import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.example.common.Constants;
import com.example.common.PageResult;
import com.example.common.enums.ErrorCodeEnum;
import com.example.common.enums.RoleEnum;
import com.example.exception.CustomException;
import com.example.mapper.UserMapper;
import com.example.pojo.entity.Account;
import com.example.pojo.entity.User;
import com.example.utils.AccountUtils;
import com.example.utils.TokenUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.math.BigDecimal;
import java.util.List;

/**
 * 普通用户 service 层实现类
 */
@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService {

    /**
     * 创建一个普通用户账户
     *
     * @param user 普通用户
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void add(User user) {
        // 1.判空
        if (user == null) {
            throw new CustomException(ErrorCodeEnum.PARAM_LOST_ERROR);
        }
        // 2.用户名不能为空
        if (StrUtil.isBlank(user.getUsername())) {
            throw new CustomException(ErrorCodeEnum.USER_NAME_IS_NOT_NULL);
        }
        // 3.没有昵称默认用用户名
        if (StrUtil.isBlank(user.getName())) {
            user.setName(user.getUsername());
        }
        // 4.没有密码默认用默认密码
        if (StrUtil.isBlank(user.getPassword())) {
            user.setPassword(Constants.USER_DEFAULT_PASSWORD);
        }
        // 5.角色是普通用户
        user.setRole(RoleEnum.USER.name());
        // 6.默认余额是 0
        user.setAccount(BigDecimal.valueOf(0));
        // 7.判断此用户名是否存在
        User dbUser = this.lambdaQuery()
                .eq(User::getUsername, user.getUsername())
                .one();
        if (dbUser != null) {
            throw new CustomException(ErrorCodeEnum.USER_EXIST_ERROR);
        }

        // 8.创建账号
        this.save(user);
    }

    /**
     * 分页查询普通用户
     *
     * @param pageNum  页码
     * @param pageSize 页大小
     * @param user  条件
     * @return 分页结果
     */
    @Override
    public PageResult<User> pageQuery(Integer pageNum, Integer pageSize, User user) {
        // 1.校验参数
        if (pageNum == null || pageNum < 1) {
            throw new CustomException(ErrorCodeEnum.PAGE_QUERY_NUM);
        }
        if (pageSize == null || pageSize < 1 || pageSize > 100) {
            throw new CustomException(ErrorCodeEnum.PAGE_QUERY_SIZE);
        }

        // 2.创建 MybatisPlus 分页对象
        IPage<User> pageObject = new Page<>(pageNum, pageSize);

        // 3.分页查询普通用户
        pageObject = this.getBaseMapper().pageQuery(pageObject, user);

        // 4.构建分页结果并返回
        return new PageResult<>(
                pageObject.getTotal(),
                pageObject.getRecords()
        );
    }

    /**
     * 根据普通用户id，查询普通用户
     *
     * @param id 普通用户id
     * @return 普通用户
     */
    @Override
    public User queryById(Integer id) {
        // 1.校验参数
        if (id == null) {
            throw new CustomException(ErrorCodeEnum.PARAM_LOST_ERROR);
        }
        // 2.根据普通用户id，查询普通用户
        User user = this.lambdaQuery()
                .eq(User::getId, id)
                .one();

        // 3.隐藏密码
        user.setPassword(null);
        return user;
    }

    /**
     * 查询全部普通用户
     *
     * @return 全部普通用户
     */
    @Override
    public List<User> queryAll() {
        // 1.查询全部普通用户
        List<User> userList = this.lambdaQuery().list();
        // 2.隐藏密码
        for (User user : userList) {
            user.setPassword(null);
        }
        return userList;
    }

    /**
     * 修改普通用户
     *
     * @param user 普通用户
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void update(User user) {
        // 1.校验参数
        if (user == null) {
            throw new CustomException(ErrorCodeEnum.PARAM_LOST_ERROR);
        }
        if (user.getId() == null) {
            throw new CustomException(ErrorCodeEnum.PARAM_LOST_ERROR);
        }

        // 2.根据普通用户id，修改普通用户
        this.updateById(user);
    }

    /**
     * 登录
     *
     * @param account 账号
     * @return 账号
     */
    @Override
    public Account login(Account account) {
        // 1.校验账号是否存在
        User dbUser = this.lambdaQuery()
                .eq(User::getUsername, account.getUsername())
                .one();
        if (dbUser == null) {
            throw new CustomException(ErrorCodeEnum.USER_NOT_EXIST_ERROR);
        }
        // 2.校验密码
        if (!dbUser.getPassword().equals(account.getPassword())) {
            throw new CustomException(ErrorCodeEnum.USER_PASSWORD_ERROR);
        }

        // 3.创建 jwt token
        // token 内容 = 账号id-角色，如 23-ADMIN
        // 密钥 = 账号密码
        String token = TokenUtils.createToken(dbUser.getId() + "-" + dbUser.getRole(), dbUser.getPassword());
        // 4.封装账号
        // 4.1 token存入对象
        dbUser.setToken(token);
        // 4.2 不返回密码
        dbUser.setPassword(null);
        // 5.返回账号
        return dbUser;
    }

    /**
     * 修改账号密码
     *
     * @param account 账号
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updatePassword(Account account) {
        // 1.根据用户名称校验账号是否存在
        User dbUser = this.lambdaQuery()
                .eq(User::getUsername, account.getUsername())
                .one();
        if (dbUser == null) {
            throw new CustomException(ErrorCodeEnum.USER_NOT_EXIST_ERROR);
        }
        // 2.根据用户输入的原密码校验与数据库的密码是否一致
        if (!dbUser.getPassword().equals(account.getPassword())) {
            throw new CustomException(ErrorCodeEnum.PARAM_OLD_PASSWORD_ERROR);
        }
        // 3.将用户输入的新密码存入数据库
        this.lambdaUpdate()
                .eq(User::getUsername, account.getUsername())
                .set(User::getPassword, account.getNewPassword())
                .update();
    }

    /**
     * 充值
     *
     * @param amount 充值的余额
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void rechargeAccount(BigDecimal amount) {
        // 1.校验余额
        if (amount == null || amount.compareTo(BigDecimal.ZERO) < 1) {
            throw new CustomException(ErrorCodeEnum.USER_RECHARGE_AMOUNT_ERROR);
        }
        // 2.获取当前用户
        Account currentLoginAccount = AccountUtils.getCurrentLoginAccount();
        if (currentLoginAccount == null) {
            throw new CustomException(ErrorCodeEnum.GET_CUR_USER_ERROR);
        }
        User user = this.lambdaQuery()
                .eq(User::getUsername, currentLoginAccount.getUsername())
                .one();
        if (user == null) {
            throw new CustomException(ErrorCodeEnum.USER_NOT_EXIST_ERROR);
        }
        // 3.根据用户 id 修改余额
        this.lambdaUpdate()
                .eq(User::getId, user.getId())
                .set(User::getAccount, user.getAccount().add(amount))
                .update();
    }
}
